from torch_geometric.graphgym.register import head_dict

from graphgps.head.inductive_node import GNNInductiveNodeHead

ssss=GNNInductiveNodeHead

# 打印已经注册的任务头
al=list(head_dict.keys())
print("Registered heads:", al)